import torch
t = torch.tensor([[[-1.2836e+00, -6.1123e-02, 1.1613e+00],
                   [-1.2836e+00, -6.1123e-02, 1.1613e+00],
                   [-1.2836e+00, -6.1123e-02, 1.1613e+00]],
                  [[-1.1613e+00, 6.1123e-02, 1.2836e+00],
                   [-1.1613e+00, 6.1123e-02, 1.2836e+00],
                   [-1.1613e+00, 6.1123e-02, 1.2836e+00]]])
print(t.mean())

t2 = torch.tensor([[[-1.2254e+00, -6.1222e-04, 1.2241e+00],
                    [-1.2254e+00, -6.1222e-04, 1.2241e+00],
                    [-1.2254e+00, -6.1222e-04, 1.2241e+00]],
                   [[-1.2241e+00, 6.1260e-04, 1.2254e+00],
                    [-1.2241e+00, 6.1260e-04, 1.2254e+00],
                    [-1.2241e+00, 6.1260e-04, 1.2254e+00]]])
print(t2.mean())